5f9c39dd65d1b6ee295e42e90b2b981d3961a2d7,JSAT/test/jsat/classifiers/trees/DecisionTreeTest.java,DecisionTreeTest,testTrainC_RegressionDataSet,#,71
Before Change
RegressionModelEvaluation rme = new RegressionModelEvaluation(instance, train);
if(useCatFeatures)
rme.setDataTransformProcess(new DataTransformProcess(new NumericalToHistogram.NumericalToHistogramTransformFactory()));
rme.evaluateTestSet(test);
assertTrue(rme.getMeanError() <= test.getTargetValues().mean()*3);
}
After Change
RegressionModelEvaluation rme = new RegressionModelEvaluation(instance, train);
if(useCatFeatures)
rme.setDataTransformProcess(new DataTransformProcess(new NumericalToHistogram.NumericalToHistogramTransformFactory(10)));
if(useCatFeatures)
rme.evaluateTestSet(train);
else
rme.evaluateTestSet(test);
assertTrue(rme.getMeanError() <= test.getTargetValues().mean()*3);
}